package com.example.llollama.user;

import java.util.HashMap;
import java.util.Map;

public class InMemoryChatContextStore {
    private static Map<String, ChatContext> contextStore = new HashMap<>();

    public static ChatContext getContext(String sessionId) {
        return contextStore.computeIfAbsent(sessionId, ChatContext::new);
    }

    public static void saveContext(ChatContext context) {
        contextStore.put(context.getSessionId(), context);
    }
}